package com.linyaonan.leetcode.sword_to_offer._68_2;

import java.util.LinkedList;

/**
 * 给定一个二叉树, 找到该树中两个指定节点的最近公共祖先。
 *
 * 百度百科中最近公共祖先的定义为：“对于有根树 T 的两个结点 p、q，最近公共祖先表示为一个结点 x，满足 x 是 p、q 的祖先且 x 的深度尽可能大（一个节点也可以是它自己的祖先）。”
 *
 * 例如，给定如下二叉树:  root = [3,5,1,6,2,0,8,null,null,7,4]
 *
 * 示例 1:
 *
 * 输入: root = [3,5,1,6,2,0,8,null,null,7,4], p = 5, q = 1
 * 输出: 3
 * 解释: 节点 5 和节点 1 的最近公共祖先是节点 3。
 * 示例 2:
 *
 * 输入: root = [3,5,1,6,2,0,8,null,null,7,4], p = 5, q = 4
 * 输出: 5
 * 解释: 节点 5 和节点 4 的最近公共祖先是节点 5。因为根据定义最近公共祖先节点可以为节点本身。
 *  
 *
 * 说明:
 *
 * 所有节点的值都是唯一的。
 * p、q 为不同节点且均存在于给定的二叉树中。
 *
 *
 * 来源：力扣（LeetCode）
 * 链接：https://leetcode.cn/problems/er-cha-shu-de-zui-jin-gong-gong-zu-xian-lcof
 * 著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。
 */
class Solution {

    private LinkedList<TreeNode> pList = new LinkedList<>();
    private LinkedList<TreeNode> qList = new LinkedList<>();


    /**
     * 思路一：找到pq两个节点的所有父节点与路径（包括自己），增删比较多，使用链表进行记录
     * 然后两个数组进行最近路径匹配即可
     * @param root
     * @param p
     * @param q
     * @return
     */
    public TreeNode lowestCommonAncestor(TreeNode root, TreeNode p, TreeNode q) {
        if (root == null) {
            return null;
        }

        try {
            bfsAndSave(root, p.val, pList);
        } catch (Exception e) {

        }
        try {
            bfsAndSave(root, q.val, qList);
        } catch (Exception e) {

        }

        // 找寻的节点并不在树上
        if (pList.isEmpty() || qList.isEmpty()) {
            return null;
        }

        TreeNode lastNode = root;

        while (!pList.isEmpty() && !qList.isEmpty()) {
            TreeNode pT = pList.removeFirst();
            TreeNode qT = qList.removeFirst();
            if (pT.val == qT.val) {
                lastNode = pT;
            } else {
                break;
            }
        }

        return lastNode;
    }

    private void bfsAndSave(TreeNode node, int p, LinkedList<TreeNode> pList) throws Exception {
        if (node == null) {
            return;
        } else {

            if (node.val == p) {
                pList.add(node);
                throw new RuntimeException("got it");
            } else {
                pList.add(node);
                bfsAndSave(node.left, p, pList);
                bfsAndSave(node.right, p, pList);
                pList.removeLast();
            }
        }
    }

}